#!/usr/bin/env python
# -*- coding: utf-8 -*-
import os
def touch_pip():
    os.system('if [ `whoami` != "root" ];then echo -e "\033[31m请使用root用户执行此脚本,退出程序！ \033[0m" && exit; else echo -e "\033[32m当前用户为root可以往下执行程序 \033[0m"; fi')
    os.system('if [ ! -d /root/.pip/ ];then mkdir -p /root/.pip/ && echo -e "\033[32m创建目录成功 \033[0m"; else echo -e "\033[33m目录存在 \033[0m"; fi')
    os.system('if [ ! -f /root/.pip/pip.conf ];then touch /root/.pip/pip.conf && echo -e "\033[32m创建主文件成功 \033[0m"; else > /root/.pip/pip.conf && echo -e "\033[32m清空文件成功 \033[0m"; fi')
    os.system("echo '[global]' > /root/.pip/pip.conf && sed -i '$a\index-url = http://mirrors.aliyun.com/pypi/simple/'  /root/.pip/pip.conf && sed -i '$a\[install]' /root/.pip/pip.conf && sed -i '$a\ttrusted-host=mirrors.aliyun.com' /root/.pip/pip.conf && echo -e '\033[32m写入成功 \033[0m'")
if __name__ == '__main__':
    touch_pip()